package cn.iocoder.yudao.module.datasource.enums;

import cn.iocoder.yudao.framework.common.core.ArrayValuable;
import lombok.AllArgsConstructor;
import lombok.Getter;

import java.util.Arrays;

/**
 * 数据源类型枚举
 * 支持多种数据库类型，包含数据库驱动类名映射
 *
 * @author 芋道源码
 */
@Getter
@AllArgsConstructor
public enum DataSourceTypeEnum implements ArrayValuable<String> {

    /**
     * MySQL 数据库
     */
    MYSQL("MYSQL", "com.mysql.cj.jdbc.Driver", "MySQL数据库"),

    /**
     * Oracle 数据库
     */
    ORACLE("ORACLE", "oracle.jdbc.driver.OracleDriver", "Oracle数据库"),

    /**
     * SQL Server 数据库
     */
    SQLSERVER("SQLSERVER", "com.microsoft.sqlserver.jdbc.SQLServerDriver", "SQL Server数据库"),

    /**
     * PostgreSQL 数据库
     */
    POSTGRESQL("POSTGRESQL", "org.postgresql.Driver", "PostgreSQL数据库"),

    /**
     * DB2 数据库
     */
    DB2("DB2", "com.ibm.db2.jdbc.app.DB2Driver", "DB2数据库"),

    /**
     * Sybase 数据库
     */
    SYBASE("SYBASE", "com.sybase.jdbc.SybDriver", "Sybase数据库"),

    /**
     * Informix 数据库
     */
    INFORMIX("INFORMIX", "com.informix.jdbc.IfxDriver", "Informix数据库"),

    /**
     * Access 数据库
     */
    ACCESS("ACCESS", "sun.jdbc.odbc.JdbcOdbcDriver", "Access数据库"),

    /**
     * 达梦数据库
     */
    DM("DM", "dm.jdbc.driver.DmDriver", "达梦数据库"),

    /**
     * 人大金仓数据库
     */
    KINGBASE("KINGBASE", "com.kingbase8.Driver", "人大金仓数据库"),

    /**
     * 华为OpenGauss数据库
     */
    OPENGAUSS("OPENGAUSS", "org.opengauss.Driver", "华为OpenGauss数据库");

    /**
     * 数据库类型标识
     */
    private final String type;

    /**
     * 数据库驱动类名
     */
    private final String driverClassName;

    /**
     * 数据库描述
     */
    private final String description;

    /**
     * 根据类型获取枚举
     *
     * @param type 数据库类型
     * @return 对应的枚举，如果不存在则返回null
     */
    public static DataSourceTypeEnum getByType(String type) {
        if (type == null) {
            return null;
        }
        
        for (DataSourceTypeEnum typeEnum : values()) {
            if (typeEnum.getType().equalsIgnoreCase(type)) {
                return typeEnum;
            }
        }
        return null;
    }

    /**
     * 根据类型获取驱动类名
     *
     * @param type 数据库类型
     * @return 驱动类名，如果类型不存在则抛出异常
     */
    public static String getDriverClassName(String type) {
        DataSourceTypeEnum typeEnum = getByType(type);
        if (typeEnum == null) {
            throw new IllegalArgumentException("不支持的数据库类型: " + type);
        }
        return typeEnum.getDriverClassName();
    }

    /**
     * 检查类型是否支持
     *
     * @param type 数据库类型
     * @return 是否支持
     */
    public static boolean isSupported(String type) {
        return getByType(type) != null;
    }

    @Override
    public String[] array() {
        return Arrays.stream(values()).map(DataSourceTypeEnum::getType).toArray(String[]::new);
    }
}